# test02_dict.py
userdb = {"zs": "123", "ls": "456"}
# 支持新用户注册(添加)，新用户名和密码注册到字典中
# 1.输入用户名,判断用户名可用性(没有在字典中出现视为可用)
# 2.如果用户名可用,要求用户输入密码
#   将用户名和密码这组键值对添加到字典userdb中
# 3.如果用户名不可用,给出对应的提示信息
def register():
    user = input("用户名: ")
    if user not in userdb:  # userdb.keys():
        pwd = input("密码: ")
        userdb[user] = pwd  # 添加
    else:
        print("用户名不可用")
# 支持老用户登陆(查询)，用户名和密码正确提示登陆成功
# 1.提示用户输入用户名和密码
# 2.用户名和密码正确打印登陆成功,否则打印登陆失败
def login():
    user = input("用户名: ")
    pwd = input("密码: ")
    if user not in userdb.keys():
        print("用户名或密码错误!")
    elif pwd != userdb.get(user):
        print("用户名或密码错误~")
    else:
        print("登陆成功")
# 3. 主程序通过循环询问，进行何种操作，根据用户的选择，执行注册或是登陆操作
def show_menu():
    while True:
        xz = input("1.register 2.login 3.exit: ")
        if xz not in ["1", "2", "3"]:
            print("请您正确输入[1/2/3]!!!!")
            continue
        elif xz == "1":
            register()  # 调用注册函数
        elif xz == "2":
            login()  # 调用登录函数
        else:  # "3"
            print("Byebye~")
            break
if __name__ == '__main__':
    show_menu()  # 调用菜单函数
